package com.lwl.client;

import com.lwl.common.JSONSerializer;
import com.lwl.common.RpcEncoder;
import com.lwl.common.RpcRequest;
import com.lwl.handler.UserClientHandler;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.string.StringDecoder;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.UUID;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * 消费者
 */
public class RPCConsumer {


    //1、创建线程池对象   处理自定义事件
    //并且当前线程池的线程数以当前系统的cpu核数
    private static ExecutorService executorService =
    Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());

    //2、声明一个自定义事件处理器  UserClientHandler

    public static UserClientHandler userClientHandler;

    //初始化客户端  创建连接池   bootStrap  甚至bootStrap  连接服务器
    public static void initClient() throws InterruptedException {
        //初始化 UserClientHandler
        userClientHandler = new UserClientHandler();
        //创建连接池对象

        NioEventLoopGroup group = new NioEventLoopGroup();
        //创建客户端引导对象
        Bootstrap bootstrap = new Bootstrap();

        //配置引导对象
        bootstrap.group(group)
                //设置对象为nio
                .channel(NioSocketChannel.class)
                //设置请求协议为tcp
                .option(ChannelOption.TCP_NODELAY,true)
                //监听handler 并初始化
                .handler(new ChannelInitializer<SocketChannel>() {
                    @Override
                    protected void initChannel(SocketChannel socketChannel) throws Exception {
                        //获取channelPipeline
                        ChannelPipeline pipeline = socketChannel.pipeline();
                        //设置编码
                        pipeline.addLast(new StringDecoder());
                        pipeline.addLast( new RpcEncoder(RpcRequest.class, new JSONSerializer()));
                        //添加自定义事件处理器
                        pipeline.addLast(userClientHandler);

                    }
                });


        bootstrap.connect("127.0.0.1",8999).sync();
    }


    //定义一个方法  使用jdk动态代理创建对象
    //serverClass 接口类型  根据哪个接口生成子类代理对象
    public static Object createProxy(final Class<?> serviceClass){
        return Proxy.newProxyInstance(Thread.currentThread().getContextClassLoader(),
                new Class[]{serviceClass}, new InvocationHandler() {

                    public Object invoke(Object o, Method method, Object[] objects) throws Throwable {

                        //1、初始化客户端client
                        if (null == userClientHandler){
                           initClient();
                        }
                        //2、给userClientHandler 设置参数
                        RpcRequest rpcRequest = new RpcRequest();
                        rpcRequest.setClassName(serviceClass.getName());
                        rpcRequest.setMethodName(method.getName());
                        rpcRequest.setParameters(objects);
                        rpcRequest.setParameterTypes(method.getParameterTypes());
                        rpcRequest.setRequestId(UUID.randomUUID().toString());

                        userClientHandler.setParam(rpcRequest);
                        //3、使用线程池  开启一个线程处理call()  写操作 并返回结果
                        Object result = executorService.submit(userClientHandler).get();
                        //返回结果
                        return result;
                    }
                }
        );
    }

}
